from dataclasses import dataclass
from typing import Dict
from pathlib import Path

@dataclass
class DataConfig:
    column_mappings: Dict[str, str] = None
    market_data_path: Path = Path("data/market")
    
    def __post_init__(self):
        if self.column_mappings is None:
            self.column_mappings = {
                "日期": "Date",
                "开盘": "Open",
                "收盘": "Close",
                "最高": "High",
                "最低": "Low",
                "成交量": "Volume",
                "成交额": "Amount"
            }

@dataclass
class Config:
    data: DataConfig = DataConfig() 